from pydantic import BaseModel
from typing import Annotated, List, Optional, Union, Iterator
from llmapi.base import *
from llmapi.config import LLMAPIConfig

class LLMAPIContext(BaseModel):
    models: Dict[Annotated[str, "model name"],Union[ChatModel, EmbeddingModel]] = {}
    config: Optional[LLMAPIConfig]
    class Config:
        arbitrary_types_allowed = True